1   /*
2    * Copyright (C) 2008 The Guava Authors
3    *
4    * Licensed under the Apache License, Version 2.0 (the "License");
5    * you may not use this file except in compliance with the License.
6    * You may obtain a copy of the License at
7    *
8    * http://www.apache.org/licenses/LICENSE-2.0
9    *
10   * Unless required by applicable law or agreed to in writing, software
11   * distributed under the License is distributed on an "AS IS" BASIS,
12   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13   * See the License for the specific language governing permissions and
14   * limitations under the License.
15   */
16  
17  package com.google.common.collect.testing.google;
18  
19  import com.google.common.annotations.GwtCompatible;
20  import com.google.common.collect.BiMap;
21  import com.google.common.collect.ImmutableBiMap;
22  import com.google.common.collect.Maps;
23  
24  import java.util.Map;
25  import java.util.Map.Entry;
26  
27  /**
28   * Generators of various {@link com.google.common.collect.BiMap}s and derived
29   * collections.
30   *
31   * @author Jared Levy
32   * @author Hayward Chan
33   */
34  @GwtCompatible
35  public class BiMapGenerators {
36    public static class ImmutableBiMapGenerator extends TestStringBiMapGenerator {
37      @Override protected BiMap<String, String> create(Entry<String, String>[] entries) {
38        ImmutableBiMap.Builder<String, String> builder = ImmutableBiMap.builder();
39        for (Entry<String, String> entry : entries) {
40          builder.put(entry.getKey(), entry.getValue());
41        }
42        return builder.build();
43      }
44    }
45    
46    public static class ImmutableBiMapCopyOfGenerator extends TestStringBiMapGenerator {
47      @Override protected BiMap<String, String> create(Entry<String, String>[] entries) {
48        Map<String, String> builder = Maps.newLinkedHashMap();
49        for (Entry<String, String> entry : entries) {
50          builder.put(entry.getKey(), entry.getValue());
51        }
52        return ImmutableBiMap.copyOf(builder);
53      }
54    }
55  }